import pyE17
from scipy import ndimage

p = pyE17.utils.imload('pinhole.png')
# Let's try this: same probe, no propagation

# ... But on a smaller array
p = ndimage.zoom(p, 192. / 256)[32:-32, 32:-32]
p = pyE17.prop.free_nf(p, 2e-10, 0.7e-3, 20e-9)

pyE17.io.h5write('probe_2_128.h5', probe=p) 
